import mysql.connector

# 用以快速的链接MySQL数据库
# 使用方法:
'''
mydb = linkMySQL(_name_) # 连接到指定的数据库，这里的_name_是数据库名
mypython = mydb.cursor()  # 创建游标对象
mypython.execute("你的SQL语句")  # 执行SQL语句
'''


# 连接MySQL到指定的数据库的函数,接受参数database_name，默认值为"python"，目前数据库用户名和数据库名都是一样的。

def linkMySQL(database_name='python', user='python'):
    mydb = mysql.connector.connect(
        host="localhost",  # 数据库主机地址
        user=str(user),  # 数据库用户名
        password="123456",  # 数据库密码
        database=str(database_name),  # 数据库名
        port="3308"  # 数据库端口，允许是字符串的形式
    )
    # 检查连接是否成功
    if mydb.is_connected():
        return mydb
    else:
        print("连接失败")
